package SubString;

import java.util.HashMap;
import java.util.Map;

/**
 * 560.和为K的子数组
 * 给你一个整数数组 nums 和一个整数 k ，请你统计并返回 该数组中和为 k 的子数组的个数 。
 * 子数组是数组中元素的连续非空序列。
 */
public class subarraySum {
    /**
     * hot100一刷
     */
    public int subarraySum(int[] nums, int k) {
        int[] preSum = new int[nums.length+1];
        int count = 0;
        preSum[0] = 0;
        preSum[1] = nums[0];
        for (int i = 2; i < preSum.length; i++) {
            preSum[i] = preSum[i - 1] + nums[i-1];
        }
        for (int i = preSum.length - 1; i >= 0; i--) {
            for (int j = i - 1; j >= 0; j--) {
                if (preSum[i] - preSum[j] == k) {
                    count++;
                }
            }
        }
        return count;
    }

    /**
     * hot100二刷
     */
    public int subarraySum2(int[] nums, int k) {
        Map<Integer, Integer> map = new HashMap<>();
        map.put(0, 1);
        int sum = 0;
        int count = 0;
        for (int i = 0; i < nums.length; i++) {
            sum += nums[i];
            if (map.containsKey(sum - k)) {
                count += map.get(sum-k);
            }
            map.put(sum, map.getOrDefault(sum, 0) + 1);
        }
        return count;
    }
}
